package com.jackstraw.create.controller;

import com.jackstraw.create.param.RespResult;
import com.jackstraw.create.utils.VelocityUtil;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.thymeleaf.util.StringUtils;

import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 版权所有： 稻草人
 * 所在包：com.jackstraw.create.controller
 * 文件名：
 * 项目：create
 * 功能描述：
 * 修改时间：2021-04-17 11:53
 *
 * @author jackstraw
 */
@RestController
@RequestMapping(value="/template")
public class TemplateController {

    /**
     * 根据目标路劲获取模板文件
     * @param folderPath
     * @return
     */
    @RequestMapping(value="/getTemplates")
    public RespResult getTemplates(String folderPath) {
        RespResult result = null;
        try {
            if (StringUtils.isEmpty(folderPath)) {
                folderPath = System.getProperty("user.dir") + "\\src\\main\\resources\\templates";
            }
            // 初始化对应的模板引擎
            VelocityUtil.initVelocityEngine(folderPath);
            // 获取模板文件列表
            File folder = new File(folderPath);
            if (folder.exists()) {
                File[] files = folder.listFiles();
                List<Map> templateList = new ArrayList<>();
                for (File file : files) {
                    String fileName = file.getName();
                    if (fileName.endsWith("vm")) {
                        HashMap<String, Object> map = new HashMap<>(4);
                        map.put("label", fileName);
                        map.put("value", fileName);
                        templateList.add(map);
                    }
                }
                result = RespResult.success();
                result.setData(templateList);
            } else {
                result = RespResult.error();
                result.setMsg("未找到文件");
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return result;
    }
}

